/*
 * Licensed to the 南京汇海交通科技有限公司
 *
 * Copyright (c) 2025.
 * All rights reserved.
 *
 * 本软件属南京汇海交通科技有限公司所有，在未获得南京汇海交通科技有限公司正式授权情况下，任何企业和个人，不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。
 */
package com.cje.gateway.security.handler;

import com.chframework.boot.core.response.result.ApiResult;
import com.chframework.boot.core.response.status.ApiStatusEnum;
import com.chframework.json.JacksonHelper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.security.web.server.authorization.ServerAccessDeniedHandler;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

/**
 * 鉴权管理，用户访问API时没有访问权限或抛出异常时的处理程序
 */
@Slf4j
@Component
public class DefaultAccessDeniedHandler implements ServerAccessDeniedHandler {

    @Override
    public Mono<Void> handle(ServerWebExchange exchange, AccessDeniedException exception) {
        return Mono.defer(() -> ApiResult.create()
                .doFirst(() -> log.error("发生{}异常！原因是：{}", exception.getClass(), exception.getMessage(), exception))
                .map(result -> result.setStatus(ApiStatusEnum.ERR_A0312))
                .map(JacksonHelper::to)
                .flatMap(message -> {
                    ServerHttpResponse response = exchange.getResponse();
                    response.setStatusCode(HttpStatus.OK);
                    // 此处需要注意的一点是，外部在后端调用本系统的接口时，
                    // 因为在此处修改了HTTP响应状态码，所以客户端无法收到响应报文，只会得到一个被抛出的异常。
                    // 基于这个考虑，此处依然需要返回200的HTTP状态码
                    //response.setStatusCode(HttpStatus.UNAUTHORIZED);
                    response.getHeaders().setContentType(MediaType.APPLICATION_JSON);
                    DataBuffer dataBuffer = response.bufferFactory().wrap(message.getBytes());
                    return response.writeWith(Mono.just(dataBuffer));
                }));
    }

}
